#include <47func.h>
int main(int argc, char *argv[])
{
    // ./chatroom_epoll  192.168.118.128 1234
    ARGS_CHECK(argc,3);
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    // AF_INET ipv4 SOCK_STREAM TCP
    struct sockaddr_in serverAddr;//带in 是ipv4地址 带in6是ipv6地址 不带是通用地址
    serverAddr.sin_family = AF_INET;//ipv4
    serverAddr.sin_addr.s_addr = inet_addr(argv[1]);//把点分十进制的ip地址转换成32位大端存储的地址
    serverAddr.sin_port = htons(atoi(argv[2]));//把端口用大端存储
    int reuse = 1;// setsockopt函数中SO_REUSEADDR的选项
    int ret = setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse));
    ERROR_CHECK(ret,-1,"setsockopt");
    ret = bind(sockfd,(struct sockaddr *)&serverAddr,sizeof(serverAddr));
    ERROR_CHECK(ret,-1,"bind");
    ret = listen(sockfd,10);
    ERROR_CHECK(ret,-1,"listen");
    struct sockaddr_in clientAddr;
    socklen_t clientAddrSize = sizeof(clientAddr);//一定要赋予初值
    int epfd = epoll_create(1); // 创建epoll文件对象
    struct epoll_event event;
    event.events = EPOLLIN;
    event.data.fd = sockfd;
    epoll_ctl(epfd,EPOLL_CTL_ADD,sockfd,&event); //把sockfd加入监听
    int netfds[4] = {-1,-1,-1,-1};
    int curidx = 0;
    char buf[4096] = {0};
    struct epoll_event readyset[10];
    while(1){
        int readynum = epoll_wait(epfd,readyset,10,-1);
        for(int i = 0; i < readynum; ++i){
            if(readyset[i].data.fd == sockfd){
                // 有客户端 connect
                netfds[curidx] = accept(sockfd,(struct sockaddr *)&clientAddr,&clientAddrSize);
                ++curidx;
                // sockfd专门用来建立连接 netfd负责和客户端的数据通信
                printf("addrlen = %d\n",clientAddrSize);
                printf("client sin_family = %d\n",clientAddr.sin_family);
                printf("client ip = %s\n", inet_ntoa(clientAddr.sin_addr));
                printf("client port =%d\n", ntohs(clientAddr.sin_port));
                // 修改监听的文件描述符
                event.events = EPOLLIN;
                event.data.fd = netfds[curidx-1];
                epoll_ctl(epfd,EPOLL_CTL_ADD,netfds[curidx-1],&event);
                if(curidx >= 4){
                    event.events = EPOLLIN;
                    event.data.fd = sockfd;
                    epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,&event);
                }
            }
            else{
                bzero(buf,sizeof(buf));
                recv(readyset[i].data.fd,buf,sizeof(buf),0);
                for(int j = 0; j < curidx; ++j){
                    if(readyset[i].data.fd == netfds[j]){
                        continue;
                    }
                    send(netfds[j],buf,strlen(buf),0);
                }
            }
        }
    }
    return 0;
}

